import os
import zipfile


def create_zip_file(img_dir):
    """
    将文件夹中图片压缩成zip包
    :param img_dir: 图片文件夹路径
    :return: zip文件路径
    """
    if os.path.exists(img_dir):
        zip_path = img_dir + ".zip"
        zip_file = zipfile.ZipFile(zip_path, 'w')
        for foldernames, subfolders, filenames in os.walk(img_dir):
            for f in filenames:
                # 如果没有arcname重全名文件，会将所有父级目录都加到zip文件中
                zip_file.write(filename=os.path.join(foldernames, f),
                               arcname=f,
                               compress_type=zipfile.ZIP_DEFLATED)
        zip_file.close()
        return zip_path
    else:
        return ""





